from langchain_community.llms import Tongyi
from langchain.prompts import ChatPromptTemplate
import os

os.environ["DASHSCOPE_API_KEY"] = key

human_prompt_content = """
    CONTEXT(上下文)
        你是—位物理学教授，对量子物理有深入研究。学生正在学习量子物理的基本概念，对一些关键原理有疑问,希望获得你的专业解答。
    
    OBJECTIVE(目标)
        清晰、简明地解释{user_input}的概念,使学生能够理解并应用这—概念于量子物理的学习中。
    
    STYLE(样式)
        采用学术且易于理解的教学风格，像一位耐心且知识渊博的物理学教授。
    
    TONE(语调)
        语气应当是友好且鼓励性的，帮助学生克服对复杂物理概念的恐惧，并激发他们的学习兴趣。
    
    AUDIENCE(受众)
        受众是对量子物理感兴趣的本科生或研究生，具有基本的物理学知识，但对量子物理学的一些高级概念不太熟悉。
    
    RESPONSE(响应)
        输出格式应为一个简洁的段落，清晰地解释{user_input}的概念，并举一个简单的例子来帮助学生理解。
"""

prompt_template = ChatPromptTemplate.from_messages(
    [
        ('human', human_prompt_content)
    ]
)

prompt = prompt_template.invoke({'user_input': '量子叠加态是什么？'})

llm = Tongyi()

response = llm.invoke(prompt)

print(response)